/*
 * $Id: VideoDecDXVAAllocator.h 2108 2010-07-06 10:51:13Z xhmikosr $
 *
 * (C) 2006-2010 see AUTHORS
 *
 * This file is part of mplayerc.
 *
 * Mplayerc is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * Mplayerc is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */


#pragma once

#include <dxva.h>
#include <dxva2api.h>


class CMPCVideoDecFilter;
class CVideoDecDXVAAllocator;


interface __declspec(uuid("AE7EC2A2-1913-4a80-8DD6-DF1497ABA494"))
IMPCDXVA2Sample :
public IUnknown
{
	STDMETHOD_(int, GetDXSurfaceId()) = 0;
};

class CDXVA2Sample : public CMediaSample, public IMFGetService, public IMPCDXVA2Sample
{
	friend class CVideoDecDXVAAllocator;

public:

	CDXVA2Sample(CVideoDecDXVAAllocator *pAlloc, HRESULT *phr);

	//Note: CMediaSample does not derive from CUnknown, so we cannot use the
	//		DECLARE_IUNKNOWN macro that is used by most of the filter classes.

	STDMETHODIMP			QueryInterface(REFIID riid, __deref_out void **ppv);
	STDMETHODIMP_(ULONG)	AddRef();
	STDMETHODIMP_(ULONG)	Release();

	// IMFGetService::GetService
	STDMETHODIMP GetService(REFGUID guidService, REFIID riid, LPVOID *ppv);

	// IMPCDXVA2Sample
	STDMETHODIMP_(int) GetDXSurfaceId();

	// Override GetPointer because this class does not manage a system memory buffer.
	// The EVR uses the MR_BUFFER_SERVICE service to get the Direct3D surface.
	STDMETHODIMP GetPointer(BYTE ** ppBuffer);

private:

	// Sets the pointer to the Direct3D surface.
	void SetSurface(DWORD surfaceId, IDirect3DSurface9 *pSurf);

	CComPtr<IDirect3DSurface9>	m_pSurface;
	DWORD						m_dwSurfaceId;
};



class CVideoDecDXVAAllocator : public CBaseAllocator
{
public:
	CVideoDecDXVAAllocator(CMPCVideoDecFilter* pVideoDecFilter, HRESULT* phr);
	virtual ~CVideoDecDXVAAllocator();

//	STDMETHODIMP GetBuffer(__deref_out IMediaSample **ppBuffer,		// Try for a circular buffer!
//                          __in_opt REFERENCE_TIME * pStartTime,
//                          __in_opt REFERENCE_TIME * pEndTime,
//                          DWORD dwFlags);
//
//	STDMETHODIMP ReleaseBuffer(IMediaSample *pBuffer);
//	CAtlList<int>			m_FreeSurface;


protected:
	HRESULT		Alloc(void);
	void		Free(void);


private :
	CMPCVideoDecFilter*		m_pVideoDecFilter;

	IDirect3DSurface9**		m_ppRTSurfaceArray;
	UINT					m_nSurfaceArrayCount;

};
